package com.beyondbit.demo.converter;

import org.apache.commons.lang3.StringUtils;
import org.springframework.core.convert.converter.Converter;
import org.springframework.stereotype.Component;

import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Date;

@Component
public class DateFormatConverter implements Converter<String, Date> {

    private static final String[] ARRAY_STRING_DATE = {"yyyy-MM", "yyyy-MM-dd", "yyyy-MM-dd HH:mm", "yyyy-MM-dd HH:mm:ss"};

    @Override
    public Date convert(String s) {
        if (StringUtils.isBlank(s)) {
            return null;
        }
        if (s.matches("^\\d{4}-\\d{1,2}$")) {
            return parseDate(s, ARRAY_STRING_DATE[0]);
        } else if (s.matches("^\\d{4}-\\d{1,2}-\\d{1,2}$")) {
            return parseDate(s, ARRAY_STRING_DATE[1]);
        } else if (s.matches("^\\d{4}-\\d{1,2}-\\d{1,2} {1}\\d{1,2}:\\d{1,2}$")) {
            return parseDate(s, ARRAY_STRING_DATE[2]);
        } else if (s.matches("^\\d{4}-\\d{1,2}-\\d{1,2} {1}\\d{1,2}:\\d{1,2}:\\d{1,2}$")) {
            return parseDate(s, ARRAY_STRING_DATE[3]);
        }

        return null;
    }

    /**
     * 根据传入的模板格式化字符串日期
     *
     * @param dateStr 字符串日期
     * @param format  模板
     * @return Date
     */
    private Date parseDate(String dateStr, String format) {
        Date date = null;
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat(format);

        try {
            date = simpleDateFormat.parse(dateStr);
        } catch (ParseException e) {
            e.printStackTrace();
        }

        return date;
    }

}
